BigQueryHook.run_with_configuration creates a new job and receives a response with location field. It should use that field for the keep_polling_job loop. Otherwise, it returns 404 job not found if the location is not US or EU.

After inserting new job and get the resp of the job details, it should retrieve the location and use it for the jobs.get request.

Make sure you have checked all steps below.

Jira

• My PR addresses the following Airflow-3874 issues and references them in the PR title. For example, "[AIRFLOW-XXX] My Airflow PR"

Description

• Here are some details about my PR, including screenshots of any UI changes:

Tests

• My PR adds the following unit tests OR does not need testing for this extremely good reason:
  test_bigquery_hook.py::TestBigQueryHookRunWithConfiguration

Commits

• My commits all reference Jira issues in their subject lines, and I have squashed multiple commits if they address the same issue. In addition, my commits follow the guidelines from "How to write a good git commit message":
  1. Subject is separated from body by a blank line
  2. Subject is limited to 50 characters (not including Jira issue reference)
  3. Subject does not end with a period
  4. Subject uses the imperative mood ("add", not "adding")
  5. Body wraps at 72 characters
  6. Body explains "what" and "why", not "how"

Documentation

• In case of new functionality, my PR adds documentation that describes how to use it.
  • When adding new operators/hooks/sensors, the autoclass documentation generation needs to be added.
  • All the public functions and the classes in the PR contain docstrings that explain what it does

Code Quality

• Passes flake8